# dataset.py
import torch
from torch.utils.data import Dataset

class DummyDataset(Dataset):
    """
    Placeholder dataset for demonstration
    
    Args:
        size (int): Number of samples in dataset (default=1000)
    """
    
    def __init__(self, size=1000):
        self.data = torch.randn(size, 3, 28, 28)
        self.labels = torch.randint(0, 10, (size,))
    
    def __len__(self):
        return len(self.data)
    
    def __getitem__(self, idx):
        return self.data[idx], self.labels[idx]
